# 学员信息在线管理

from stu_sql import stu_sql
# 定义一个用于存放学员信息的列表变量
'''
stu 表的字段有
id int(2),主键，唯一，自增
name varchar
age int(2) 默认20
classid varchar
'''

stu = stu_sql()
ll = stu.findAll()
# 定义一个学生信息的输出函数
def showStu():
    stu = stu_sql()
    ll = stu.findAll()
    if len(ll) == 0:
        print("========== 没有学员信息可以输出！=============")
        return
    print("|{0:<5}| {1:<10}| {2:<5}| {3:<10}|".format("sid", "name", "age", "classid"))
    print("-" * 40)
    for i in ll:
        print("|{0:<5}| {1:<10}| {2:<5}| {3:<10}|".format(i[0],i[1],i[2],i[3]))


while True:
    # 输出初始界面
    print("=" * 12, "学员管理系统", "=" * 14)
    print("{0:1} {1:13} {2:15}".format(" ", "1. 查看学员信息", "2. 添加学员信息"))
    print("{0:1} {1:13} {2:15}".format(" ", "3. 删除学员信息", "4. 退出系统"))
    print("=" * 40)
    key = input("请输入对应的选择：")
    # 根据键盘值，判断并执行对应的操作
    if key == "1":
        print("=" * 12, "学员信息浏览", "=" * 14)
        showStu()
        input("按回车键继续：")
    elif key == "2":
        print("=" * 12, "学员信息添加", "=" * 14)

        stu1 = ['','','']
        stu1[0] = input("请输入要添加的姓名：")
        stu1[1] = int(input("请输入要添加的年龄："))
        stu1[2] = input("请输入要添加的班级号：")
        stu.insert(stu1)
        showStu()
        input("按回车键继续：")
    elif key == "3":
        print("=" * 12, "学员信息删除", "=" * 14)
        showStu()
        sid = input("请输入你要删除的信息id号：")
        stu.delete(int(sid))
        showStu()
        input("按回车键继续：")
    elif key == "4":
        print("=" * 12, "再见", "=" * 14)
        break
    else:
        print("======== 无效的键盘输入！ ==========")
